cd ../..

gpus=$1

# Initialize CNN version
for freeze_layer in 3
do
    echo "Freeze CNN layer ${freeze_layer}"
    python joint_main.py \
        --gpus ${gpus}, \
        --opt_conf ./configs/opt_joint.yaml \
        --exp_name joint/OOD_CRNN_w1.0_wMT_o${freeze_layer}CNN+RNN-cosinedown \
        --feat_dim 256 \
        --grad_compensate False \
        --joint_itpl_warmup 80 \
        --ood_weight 1.0 \
        --unfreeze_cnn_layer ${freeze_layer} 
done
#### In this code, `total_loss = id_loss + od_loss * self.opt_configs["ood_weight"] * itpl_lambda`